CUDA_VISIBLE_DEVICES=1,2,3 python -m torch.distributed.launch --use_env --nproc_per_node=3 main.py \
--model_name "MIMO-UNetPlus" \
--mode "train" \
--batch_size 2 \
--num_epoch 500 \
--save_freq 50 \
--valid_freq 1 \
--print_freq 10 \
--exp_name "exp_test" \
--data_dir "./CamVid"
